Multiple Return Statement in a Python Function